pattern drawing in python